#include <linux/kernel.h>
#include <linux/delay.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/errno.h>
#include <asm/mach/map.h>
#include <asm/uaccess.h>
#include <asm/io.h>
#include <linux/gpio.h>
#include <linux/ide.h>
#include <linux/types.h> 

#include <linux/device.h>
#include <linux/cdev.h>
#include <linux/of.h>
#include <linux/of.h>
#include <linux/of_address.h>
#include <linux/of_irq.h>

#define DTSLED_CNT    1
#define DTSLED_NAME   "dtsled"
#define LEDOFF        0
#define LEDON         1


/* 映射后的寄存器虚拟地址指针 */
static void __iomem *IMX6U_CCM_CCGR1;
static void __iomem *SW_MUX_GPIO1_IO04;
static void __iomem *SW_PAD_GPIO1_IO04;
static void __iomem *GPIO1_DR;
static void __iomem *GPIO1_GDIR;

/*dtsled设备结构体*/
struct dtsled_dev{
    dev_t devid;              //设备号
    struct cdev cdev;         
    struct class *class;
    struct device *device;
    int major;
    int minor;
    struct device_node *nd;
};
struct dtsled_dev dtsled;


/*
 * @description		: 打开设备
 * @param - inode 	: 传递给驱动的inode
 * @param - filp 	: 设备文件，file结构体有个叫做private_data的成员变量
 * 					  一般在open的时候将private_data指向设备结构体。
 * @return 			: 0 成功;其他 失败
 */
static int led_open(struct inode *inode,struct file *filp)
{
    filp->private_data = &dtsled; //设置私有数据
    return 0;
}

/*
 * @description		: 从设备读取数据 
 * @param - filp 	: 要打开的设备文件(文件描述符)
 * @param - buf 	: 返回给用户空间的数据缓冲区
 * @param - cnt 	: 要读取的数据长度
 * @param - offt 	: 相对于文件首地址的偏移
 * @return 			: 读取的字节数，如果为负值，表示读取失败
 */
static ssize_t led_read(struct file *filp, char __user *buf, size_t cnt, loff_t *offt)
{
    return 0;
}

void led_switch(u8 sta)
{
	u32 val = 0;
	if(sta == LEDON) {
		val = readl(GPIO1_DR);
		val &= ~(1 << 4);	
		writel(val, GPIO1_DR);
	}else if(sta == LEDOFF) {
		val = readl(GPIO1_DR);
		val|= (1 << 4);	
		writel(val, GPIO1_DR);
	}	
}
/*
 * @description		: 向设备写数据 
 * @param - filp 	: 设备文件，表示打开的文件描述符
 * @param - buf 	: 要写给设备写入的数据
 * @param - cnt 	: 要写入的数据长度
 * @param - offt 	: 相对于文件首地址的偏移
 * @return 			: 写入的字节数，如果为负值，表示写入失败
 */
static ssize_t led_write(struct file *filp, const char __user *buf, size_t cnt, loff_t *offt)
{
    int retvalue;
    unsigned char databuf[1];
    unsigned char ledstate;

    retvalue = copy_from_user(databuf,buf,cnt);
    if(retvalue<0)
    {
        printk("kernel write failed!\r\n");
        return -EFAULT;
    }

    ledstate = databuf[0];

	if(ledstate == LEDON) {	
		led_switch(LEDON);		/* 打开LED灯 */
	} else if(ledstate == LEDOFF) {
		led_switch(LEDOFF);	/* 关闭LED灯 */
	}

    return 0;
}


/*
 * @description		: 关闭/释放设备
 * @param - filp 	: 要关闭的设备文件(文件描述符)
 * @return 			: 0 成功;其他 失败
 */
static int led_release(struct inode *inode, struct file *filp)
{
	return 0;
}


/* 设备操作函数 */
static struct file_operations dtsled_fops = {
    .owner = THIS_MODULE,
    .open = led_open,
    .write = led_write,
    .read = led_read,
    .release = led_release,
};

static int __init led_init(void)
{
    struct property *proper;
    int ret;
    const char *str;
    u32 regdata[14];
    u32 val=0;
    
	/* 1、获取设备节点：debianled */
    dtsled.nd = of_find_node_by_path("/debianled");
    if(dtsled.nd==NULL)
    {
        printk("debianled node nost find!\r\n");
        return -EINVAL;
    }else{
		printk("debianled node find!\r\n");
    }

	/* 2、获取compatible属性内容 */
    proper = of_find_property(dtsled.nd,"compatible",NULL);
    if(proper==NULL)
    {
    printk("compatible property find failed\r\n");   
    }\
    else{
    printk("compatible = %s\r\n", (char*)proper->value);
    }

	/* 3、获取status属性内容 */
    ret = of_property_read_string(dtsled.nd,"status",&str);
 	if(ret < 0){
		printk("status read failed!\r\n");
	} else {
		printk("status = %s\r\n",str);
	}   
	/* 4、获取reg属性内容 */
    ret = of_property_read_u32_array(dtsled.nd,"reg",regdata,10);
    if(ret<0)
    {
 	printk("reg property read failed!\r\n");       
    }else{
        u8 i = 0;
		printk("reg data:\r\n");        
        for(i = 0; i < 10; i++)
		printk("%#X ", regdata[i]);
		printk("\r\n");
    }


 	/* 初始化LED */
    /* 1、寄存器地址映射 */
 	IMX6U_CCM_CCGR1 = of_iomap(dtsled.nd, 0);
	SW_MUX_GPIO1_IO04 = of_iomap(dtsled.nd, 1);
  	SW_PAD_GPIO1_IO04 = of_iomap(dtsled.nd, 2);
	GPIO1_DR = of_iomap(dtsled.nd, 3);
	GPIO1_GDIR = of_iomap(dtsled.nd, 4);    

	/* 2、使能GPIO1时钟 */
	val = readl(IMX6U_CCM_CCGR1);
	val &= ~(3 << 26);	//将26 27位清零
	val |= (3 << 26);	//将26 27位设置1
    writel(val, IMX6U_CCM_CCGR1);

	/* 3、设置GPIO1_IO04的复用功能，将其复用为
	 *    GPIO1_IO04，最后设置IO属性。
	 */
	writel(5, SW_MUX_GPIO1_IO04);

	/*寄存器SW_PAD_GPIO1_IO03设置IO属性
	 *bit 16:0 HYS关闭
	 *bit [15:14]: 00 默认下拉
     *bit [13]: 0 kepper功能
     *bit [12]: 1 pull/keeper使能
     *bit [11]: 0 关闭开路输出
     *bit [7:6]: 10 速度100Mhz
     *bit [5:3]: 110 R0/6驱动能力
     *bit [0]: 0 低转换率
	 */
	writel(0x10B0, SW_PAD_GPIO1_IO04);

	/* 4、设置GPIO1_IO04为输出功能 */
	val = readl(GPIO1_GDIR);
	val &= ~(1 << 4);	/* 清除以前的设置 */
	val |= (1 << 4);	/* 设置为输出 */
	writel(val, GPIO1_GDIR);

	/* 5、默认关闭LED */
	val = readl(GPIO1_DR);
	val |= (1 << 4);	
	writel(val, GPIO1_DR);

	/* 1、创建设备号 */
    if(dtsled.major)
    {
        dtsled.devid = MKDEV(dtsled.major,0);
        register_chrdev_region(dtsled.devid,DTSLED_CNT,DTSLED_NAME);
    }else{
        alloc_chrdev_region(&dtsled.devid,0,DTSLED_CNT,DTSLED_NAME);
        dtsled.major = MAJOR(dtsled.devid);
        dtsled.minor = MINOR(dtsled.devid);
    }
	printk("dtsled major=%d,minor=%d\r\n",dtsled.major, dtsled.minor);	

	/* 2、初始化cdev */
    dtsled.cdev.owner = THIS_MODULE;
    cdev_init(&dtsled.cdev,&dtsled_fops);

    /* 3、添加一个cdev */
    cdev_add(&dtsled.cdev,dtsled.devid,DTSLED_CNT);

    /* 4、创建类 */
 	dtsled.class = class_create(THIS_MODULE, DTSLED_NAME);  
    if (IS_ERR(dtsled.class)) {
		return PTR_ERR(dtsled.class);
	} 

 	/* 5、创建设备 */   
    dtsled.device = device_create(dtsled.class, NULL, dtsled.devid, NULL, DTSLED_NAME);
  	if (IS_ERR(dtsled.device)) {
		return PTR_ERR(dtsled.device);
	}  
    return 0;
}

static void __exit led_exit(void)
{

	/* 注销字符设备驱动 */
	cdev_del(&dtsled.cdev);/*  删除cdev */
	unregister_chrdev_region(dtsled.devid, DTSLED_CNT); /* 注销设备号 */

	device_destroy(dtsled.class, dtsled.devid);
	class_destroy(dtsled.class);
}

//注册模块入口和出口
module_init(led_init);
module_exit(led_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("JiaYi");




